package com.jflm.util;

import java.util.Random;

/**
 * ******************  类说明  *********************
 * class       :  RedpacketAlgorithmUtil
 * @author     :  hejinyun@umfintech.com
 * @version    :  1.0  
 * description :  红包分配算法
 * @see        :                        
 * ***********************************************
 */
public class RedpacketAlgorithmUtil {

	private static Random random = new Random();
	static {
		random.setSeed(System.currentTimeMillis());
	}
	
	/**
	 * ********************************************
	 * method name   : xRandom 
	 * description   : 生产min和max之间的随机数，但是概率不是平均的，从min到max方向概率逐渐加大。
	 * 先平方，然后产生一个平方值范围内的随机数，再开方，这样就产生了一种“膨胀”再“收缩”的效果。
	 * @return       : long
	 * @param        : @param min
	 * @param        : @param max
	 * @param        : @return
	 * modified      : hejinyun@umfintech.com ,  2017-6-7  下午7:58:14
	 * @see          : 
	 * *******************************************
	 */
	public static long xRandom(long min, long max) {
		return sqrt(nextLong(sqr(max - min)));
	}


	/**
	 * ********************************************
	 * method name   : generate 
	 * description   : 红包分配算法
	 * @return       : long[]
	 * @param        : @param total 红包总金额
	 * @param        : @param count 红包个数
	 * @param        : @param max 每个红包的最大额
	 * @param        : @param min 每个红包的最小额
	 * @param        : @return
	 * modified      : hejinyun@umfintech.com ,  2017-6-7  下午7:56:37
	 * @see          : 
	 * *******************************************
	 */
	public static long[] generate(long total, int count, long max, long min) {
		long[] result = new long[count];

		long average = total / count;

		//long a = average - min;
		//long b = max - min;

		//
		//这样的随机数的概率实际改变了，产生大数的可能性要比产生小数的概率要小。
		//这样就实现了大部分红包的值在平均数附近。大红包和小红包比较少。
		//long range1 = sqr(average - min);
		//long range2 = sqr(max - average);

		for (int i = 0; i < result.length; i++) {
			//因为小红包的数量通常是要比大红包的数量要多的，因为这里的概率要调换过来。
			//当随机数>平均值，则产生小红包
			//当随机数<平均值，则产生大红包
			if (nextLong(min, max) > average) {
				// 在平均线上减钱
				long temp = min + xRandom(min, average);
				result[i] = temp;
				total -= temp;
			} else {
				// 在平均线上加钱
				long temp = max - xRandom(average, max);
				result[i] = temp;
				total -= temp;
			}
		}
		// 如果还有余钱，则尝试加到小红包里，如果加不进去，则尝试下一个。
		while (total > 0) {
			for (int i = 0; i < result.length; i++) {
				if (total > 0 && result[i] < max) {
					result[i]++;
					total--;
				}
			}
		}
		// 如果钱是负数了，还得从已生成的小红包中抽取回来
		while (total < 0) {
			for (int i = 0; i < result.length; i++) {
				if (total < 0 && result[i] > min) {
					result[i]--;
					total++;
				}
			}
		}
		return result;
	}

	public static long sqrt(long n) {
		return (long) Math.sqrt(n);
	}

	public static long sqr(long n) {
		return n * n;
	}
	
	public static long nextLong(long n) {
		return random.nextInt((int) n);
	}

	static long nextLong(long min, long max) {
		return random.nextInt((int) (max - min + 1)) + min;
	}
	
	public static void main(String[] args) throws Exception{
		long max = 70;
		long min = 1;

		long[] result = RedpacketAlgorithmUtil.generate(500, 10, max, min);
		long total = 0;
		for (int i = 0; i < result.length; i++) {
			System.out.println("result[" + i + "]:" + result[i]);
			total += result[i];
		}
		//检查生成的红包的总额是否正确
		System.out.println("total:" + total);
	}
	
}
